package com.java.servlet;

import org.apache.commons.io.IOUtils;

import javax.servlet.ServletContext;
import javax.servlet.ServletException;
import javax.servlet.ServletOutputStream;
import javax.servlet.http.HttpServlet;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.io.IOException;
import java.io.InputStream;
import java.net.URLEncoder;

/**
 * @author jingjintao
 * @date 2021/6/19 - 0:29
 **/
public class DownloadServlet extends HttpServlet {

    @Override
    protected void doGet(HttpServletRequest req, HttpServletResponse resp) throws ServletException, IOException {
        //1.获取要下载的文件名
        String downloadFileName = "3.jpg";
        //2.读取要下载的内容 （通过ServletContext对象可以获取）
        ServletContext servletContext = getServletContext();
        //4.在回传前，通过响应头告诉客户端返回的数据类型
        //获取要下载的文件类型
        String mimeType = servletContext.getMimeType("/file/" + downloadFileName);
        System.out.println(mimeType);
        resp.setContentType(mimeType);
        //5.还要告诉客户端收到的数据是用于下载使用 （还是使用响应头）
        //Content-Disposition 响应头表示收到的数据怎么处理
        //attachment 表示附件，表示下载使用
        //filename= 表示制定下载的文件名
        //url编码是把汉字转换成为%xx%xx的格式
        resp.setHeader("Content-Disposition","attachment; filename = " + URLEncoder.encode("中国.jpg","UTF-8"));

        /**
         * 斜杠被服务器解析表示地址为http://ip:port/工程名  映射到代码的web目录
         */
        InputStream is = servletContext.getResourceAsStream("/file/" + downloadFileName);
        //获取响应的输出流
        ServletOutputStream os = resp.getOutputStream();
        //读取输入流中的全部数据，复制给输出流，输出给客户端
        //3.把下载的文件内容回传给客户端
        IOUtils.copy(is,os);

    }
}
